home *** CD-ROM | disk | FTP | other *** search
/ Nebula 1 / Nebula One.iso / Utilities / Unix / satan-1.1.1 / perl / services.pl < prev    next >
Text File  |  1995-04-10  |  4KB  |  169 lines
  1. #
  2. # infer_services - classify host by services used and provided. Output to: 
  3. #
  4. # $servers{service}{host}, $clients{service}{host}: Service is, for
  5. # example, "anonymous FTP". The servers (clients) table holds per service
  6. # and host, all SATAN records on that topic.
  7. #
  8. # $server_counts{service}, $client_counts{service}: host counts of the
  9. # corresponding entries in $servers and $clients.
  10. #
  11. # $server_severities{service}, $client_severities{service}: ditto, with
  12. # at least one vulnerability.
  13. #
  14. # $server_info{host}: newline-delimited list of services provided per host.
  15. # $client_info{host}: newline-delimited list of services consumed per host.
  16. #
  17. # $service_flag: reset whenever the tables are updated. To recalculate,
  18. # invoke make_service_info().
  19. #
  20. # Standalone usage: perl services.pl [satan_record_files...]
  21. # 
  22.  
  23. $services_files = "rules/services";
  24.  
  25. sub build_infer_services {
  26.     local($files) = @_;
  27.     local($service, $code, $file, $cond, $class, $host);
  28.  
  29.     $code = "sub infer_services {\n";
  30.     $code .= "\tlocal(\$type);\n";
  31.  
  32.     foreach $file (split(/\s+/, $files)) {
  33.     open(RULES, $file) || die "cannot open $file: $!";
  34.     while (<RULES>) {
  35.         chop;
  36.         while (/\\$/) {
  37.         chop;
  38.         $_ .= <RULES>;
  39.         chop;
  40.         }
  41.         s/#.*$//;
  42.         s/\s+$//;
  43.         next if /^$/;
  44.         if (/^(servers|clients)/i) {
  45.         ($class = $1) =~ tr /A-Z/a-z/;
  46.         } else {
  47.         s/@/\\@/g;
  48.         ($cond, $type, $host) = split(/\t+/, $_, 3);
  49.         die "missing service name" if $type eq "";
  50.         $host = "\$target" if $host eq "";
  51.         $code .= "\
  52.     if ($cond) {
  53.         \$$class\{\"$type\"}{$host} .= \$_ . \"\\n\";
  54.         \$service_flag = 0;
  55.     }
  56. ";
  57.         # %{${$class}{$type}} = ();
  58.         }
  59.     }
  60.     close(RULES);
  61.     }
  62.     $code .= "}\n";
  63.     return $code;
  64. }
  65.  
  66. #
  67. # Generate services-dependent statistics.
  68. #
  69. sub make_service_info {
  70.     local($service, $host, %junk);
  71.  
  72.     if ($service_flag > 0) {
  73.     return;
  74.     }
  75.     $service_flag = time();
  76.     &make_severity_info();
  77.  
  78.     print "Rebuild service type statistics...\n" if $debug;
  79.  
  80.     %server_info = ();
  81.     for $service (keys %servers) {
  82.     %junk = %{$servers{$service}};
  83.     $server_counts{$service} = sizeof(*junk);
  84.     $server_severities{$service} = 0;
  85.     for $host (keys %junk) {
  86.         $server_info{$host} .= $service . "\n";
  87.         if (exists($severity_host_type_info{$host})) {
  88.         $server_severities{$service}++;
  89.         }
  90.     }
  91.     }
  92.     %client_info = ();
  93.     for $service (keys %clients) {
  94.     %junk = %{$clients{$service}};
  95.     $client_counts{$service} = sizeof(*junk);
  96.     $client_severities{$service} = 0;
  97.     for $host (keys %junk) {
  98.         $client_info{$host} .= $service . "\n";
  99.         if (exists($severity_host_type_info{$host})) {
  100.         $client_severities{$service}++;
  101.         }
  102.     }
  103.     }
  104. }
  105.  
  106. #
  107. # Reset the service information tables.
  108. #
  109. sub clear_service_info {
  110.     %servers = ();
  111.     %clients = ();
  112.     %server_severities = ();
  113.     %client_severities = ();
  114.     %server_info = ();
  115.     %client_info = ();
  116.     $service_flag = 0;
  117. }
  118.  
  119. #
  120. # Some scaffolding for stand-alone operation
  121. #
  122. if ($running_under_satan) {
  123.     eval &build_infer_services($services_files);
  124.     die "error in $services_files: $@" if $@;
  125. } else {
  126.     $running_under_satan = -1;
  127.     $debug = 1;
  128.  
  129.     require 'perl/misc.pl';
  130.  
  131.     #
  132.     # Generate code from rules files.
  133.     #
  134.     $code = &build_infer_services($services_files);
  135.     print "Code generated from $services_files:\n\n";
  136.     print $code;
  137.     eval $code; 
  138.     die "error in $services_files: $@" if $@;
  139.  
  140.     #
  141.     # Apply rules.
  142.     #
  143.     print "\nApplying rules to all SATAN records...\n";
  144.     while (<>) {
  145.     chop;
  146.     if (&satan_split($_) == 0) {
  147.         &infer_services(_);
  148.     }
  149.     }
  150.     &make_service_info();
  151.  
  152.     print "Servers\n";
  153.     for $service (sort keys %servers) {
  154.        print "\t$service ($server_counts{$service})\n";
  155.     for (sort keys %{$servers{$service}}) {
  156.         print "\t\t$_\n";
  157.     }
  158.     }
  159.     print "Clients\n";
  160.     for $service (sort keys %clients) {
  161.        print "\t$service ($client_counts{$service})\n";
  162.     for (sort keys %{$clients{$service}}) {
  163.         print "\t\t$_\n";
  164.     }
  165.     }
  166. }
  167.  
  168. 1;
  169.